package com.zyg.controller;

import com.alibaba.fastjson2.JSON;
import com.baomidou.mybatisplus.core.conditions.query.QueryWrapper;
import com.baomidou.mybatisplus.extension.plugins.pagination.Page;
import com.zyg.pojo.Grade;
import com.zyg.service.GradeService;
import com.zyg.util.LayData;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.stereotype.Controller;
import org.springframework.ui.Model;
import org.springframework.web.bind.annotation.*;

import java.util.List;

@Controller
@RequestMapping("/grade")
public class GradeController {

    @Autowired
    private GradeService gradeService;

    @RequestMapping("/tolist")
    public String tolist() {
        return "grade/list";
    }

    @RequestMapping("/list")
    @ResponseBody
    public String list(Grade grade,
                       @RequestParam(value = "page", defaultValue = "1") Integer page,
                       @RequestParam(value = "limit", defaultValue = "10") Integer limit) {
        QueryWrapper<Grade> queryWrapper = new QueryWrapper<>();
        if (grade.getGradeName() != null && !grade.getGradeName().isEmpty()) {
            queryWrapper.like("grade_name", grade.getGradeName());
        }

        long totalCount = gradeService.count(queryWrapper);
        Page<Grade> page1 = new Page<>(page, limit);
        List<Grade> gradeList = gradeService.page(page1, queryWrapper).getRecords();

        LayData layData = new LayData();
        layData.setCount((int) totalCount);
        layData.setData(gradeList);
        layData.setCode(0);
        layData.setMsg("查询成功");

        return JSON.toJSONString(layData);
    }

    @RequestMapping("/toadd")
    public String toadd() {
        return "grade/add";
    }

    @RequestMapping("/addSave")
    @ResponseBody
    public String addSave(@ModelAttribute Grade grade) {
        try {
            boolean result = gradeService.save(grade);
            return JSON.toJSONString(new LayData(result ? 0 : 1, result ? "保存成功" : "保存失败"));
        } catch (Exception e) {
            e.printStackTrace();
            return JSON.toJSONString(new LayData(1, "保存失败: " + e.getMessage()));
        }
    }

    @RequestMapping("/toupdate/{id}")
    public String toUpdate(@PathVariable("id") int id, Model model) {
        Grade grade = gradeService.getById(id);
        model.addAttribute("grade", grade);
        return "grade/update";
    }

    @RequestMapping("/updateSave")
    @ResponseBody
    public String updateSave(Grade grade) {
        boolean result = gradeService.updateById(grade);
        return JSON.toJSONString(new LayData(result ? 0 : 500, result ? "更新成功" : "更新失败"));
    }

    @RequestMapping("/delete/{id}")
    @ResponseBody
    public String delete(@PathVariable("id") int id) {
        boolean result = gradeService.removeById(id);
        return JSON.toJSONString(new LayData(result ? 0 : 500, result ? "删除成功" : "删除失败"));
    }
}
